#include <iostream>
#include <stdio.h>
#include "diamond.h"

using namespace std;

int main(int argc, char * argv[]) {

    if (argc > 1) {
        int num = 0;
        for (int i =1; i< argc; ++i) {
            sscanf(argv[i], "%d", &num); 
            cout << Diamond(num);
        }
        return 0;
    }

    cout << Diamond(10) << endl;
    cout << Diamond(3) << endl;
    cout << Diamond(40) << endl;

    Diamond test( -1 );
    test.display();
    test.setStars(15);
    cout << test;

    return 0;
}
